#include "problems/4.h"
#include "gtest/gtest.h"

TEST(TestProblemSolutions, TestProblem4)
{
    problem_4::Solution s;
    vector<int> nums1{1, 3};
    vector<int> nums2{2};
    ASSERT_FLOAT_EQ(s.findMedianSortedArrays(nums1, nums2), 2.0);
    nums1 = {1, 2};
    nums2 = {3, 4};
    ASSERT_FLOAT_EQ(s.findMedianSortedArrays(nums1, nums2), 2.5);
}